package com.toulezu.test.component;

import freemarker.cache.StringTemplateLoader;
import freemarker.template.Configuration;
import freemarker.template.Template;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import org.springframework.ui.freemarker.FreeMarkerTemplateUtils;
import org.springframework.web.servlet.view.freemarker.FreeMarkerConfigurer;

import java.io.StringWriter;
import java.util.Map;

@Component
public class TemplateCom {


    private static Logger logger = LoggerFactory.getLogger(TemplateCom.class);

    @Autowired
    private FreeMarkerConfigurer freeMarkerConfigurer;

    /**
     * 处理模板文件
     *
     * @param template 模板文件名称
     * @param dataMap 模板数据
     * @return 解析后的内容
     */
    public String renderTemplate(String template, Map<String, Object> dataMap) {
        try {
            Template tpl = freeMarkerConfigurer.getConfiguration().getTemplate(template);
            return FreeMarkerTemplateUtils.processTemplateIntoString(tpl, dataMap);
        } catch (Exception e) {
            logger.error(this.getClass().getName().concat(".renderTemplate has error"), e);
            return null;
        }

    }

    /**
     * 处理模板字符串
     *
     * @param string 模板字符串
     * @param dataMap 模板数据
     * @return 解析后的内容
     */
    public synchronized String renderString(String string, Map<String, Object> dataMap) {
        try {
            if (string.contains("${") && string.contains("}")) {
                // 1. 获取已经存在的配置
                Configuration configuration = new Configuration(Configuration.VERSION_2_3_28);

                // 2. 自定义一个字符串模板
                StringTemplateLoader stringLoader = new StringTemplateLoader();
                stringLoader.putTemplate("myTemplate", string);

                // 3. 加载已经存在的模板
                configuration.setTemplateLoader(stringLoader);

                // 4. 处理模板数据
                StringWriter writer = new StringWriter();
                configuration.getTemplate("myTemplate").process(dataMap, writer);
                return writer.toString();
            }
            return string;
        } catch (Exception e) {
            logger.error(this.getClass().getName().concat(".renderString has error"), e);
            return null;
        }

    }

}
